package com.yww.leetcode.linkedlist;

import com.yww.leetcode.common.ListNode;

import java.util.Random;

/**
 * @author yww
 * @description 21. 合并两个有序链表
 * @since 2024/2/2 15:48
 */
public class MergeTwoLists21 {
    // 将两个升序链表合并为一个新的 升序 链表并返回。
    // 新链表是通过拼接给定的两个链表的所有节点组成的。
    public ListNode mergeTwoLists(ListNode list1, ListNode list2) {
        int i = new Random().nextInt();
        if (i % 2 == 0) {
            return merge01(list1, list2);
        } else {
            return merge02(list1, list2);
        }
    }

    // 方法1：构造一新链表，将两个两表中较小一个节点插入新链表
    private static ListNode merge01(ListNode list1, ListNode list2) {
        ListNode sentinel = new ListNode(-1, null);
        ListNode p = sentinel;
        // 循环，将两个链表小的节点插入新链表末尾
        while (list1 != null && list2 != null) {
            if (list1.val < list2.val) {
                p.next = list1;
                list1 = list1.next;
            } else {
                p.next = list2;
                list2 = list2.next;
            }
            // p 后移
            p = p.next;
        }
        // 将原链表剩余节点链到新链表末尾
        if (list1 != null) {
            p.next = list1;
        } else if (list2 != null) {
            p.next = list2;
        }
        return sentinel.next;
    }

    // 方法二：递归
    private static ListNode merge02(ListNode list1, ListNode list2) {
        // 如果其中一个链表为空，直接返回不为空的链表
        if (list1 == null) {
            return list2;
        } else if (list2 == null) {
            return list1;
        }

        // 就返回小的节点
        if (list1.val < list2.val) {
            // 返回之前，节点1.next参与下次递归,节点2不变
            list1.next = merge02(list1.next, list2);
            return list1;
        } else {
            // 返回之前，节点2.next参与下次递归,节点1不变
            list2.next = merge02(list1, list2.next);
            return list2;
        }
    }
}
